How to use word embedding in python